package com.atguigu.senior.util;

import com.alibaba.druid.pool.DruidDataSourceFactory;

import javax.sql.DataSource;
import java.io.InputStream;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.Properties;

/**
 * JDBC 工具类(V2.0)
 * 1. 维护一个连接池对象, 维护了一个线程绑定变量的ThreadLocal对象
 * 2、对外提供在ThreadLocal中获取连接的方法
 * 3、对外提供回收连接的方法，回收过程中，将要回收的连接从ThreadLocal中移除！
 * 注意 : 工具类仅对外提供共性的功能代码, 所以方法均为静态方法
 * 注意：使用ThreadLocal就是为了一个线程在多次数据库操作过程中，使用的是同一个连接！
 */
public class JDBCUtilV2 {
    /**
     * 创建连接池引用, 并设置为静态私有的.
     * 私有是因为不允许其他类访问, 静态是因为需要在静态方法中初始化
     */
    private static DataSource dataSource;
    private static ThreadLocal<Connection> threadLocal = new ThreadLocal<>();

    /**
     * 在静态代码块中完成连接池的创建
     */
    static {
        Properties properties = new Properties();
        try (InputStream resourceAsStream = JDBCUtil.class.getClassLoader().getResourceAsStream("db.properties")) {
            properties.load(resourceAsStream);
            dataSource = DruidDataSourceFactory.createDataSource(properties);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }


    /**
     * 获取连接
     *
     * @return
     * @throws SQLException
     */
    public static Connection getConnection() throws SQLException {
        // 在threadLocal获取connection对象
        Connection connection = threadLocal.get();
        // 如果connection为空, 说明是第一次获取
        if (connection == null) {
            // 在连接池中获取一个连接, 存储在threadLocal中
            connection = dataSource.getConnection();
            threadLocal.set(connection);
        }
        // 返回connection对象
        return connection;
    }

    /**
     * 回收连接
     */
    public static void release() {
        // 获取threadLocal中的connection对象
        Connection connection = threadLocal.get();
        // 如果connection不为空, 说明获取过connection连接对象, 需要回收
        if (connection != null) {
            try {
                // 如果开启了事务的手动提交, 操作完毕后, 在归还给连接池之前, 需要将事务改回自动提交
                connection.setAutoCommit(true);
            } catch (SQLException e) {
                // 处理 setAutoCommit 异常
                System.err.println("Error setting auto-commit to true: " + e.getMessage());
            }
            try {
                // 连接池回收连接
                connection.close();
            } catch (Exception e) {
                // 处理 close 异常
                System.err.println("Error closing connection: " + e.getMessage());
            } finally {
                // 从threadLocal中移除
                threadLocal.remove();
            }
        }
    }

}
